package yxxy.c_021;

import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

/**
 * 面试题：写一个固定容量同步容器，拥有put和get方法，以及getCount方法。
 * 能够支持2个生产者线程以及10个消费者线程的阻塞调用
 * <p>
 * 使用wait和notify/notifyAll来实现
 * <p>
 * 使用lock和condition来实现
 * 对比两种方式，Condition的方式可以更加精确的指定哪些线程被唤醒
 *
 * @create: 2019-08-29 21:20
 **/
public class MyCoutainer2<T> {
    //容器
    final private List<T> lists = new LinkedList<>();
    //容器的最大数量
    final private int MAX = 10;

    //容器内当前物品的数量
    private int count = 0;

    //可以重入锁(手动锁)
    private Lock lock = new ReentrantLock();
    //condition：条件，可以作为线程的条件来使用，类似于精确指定线程为某个类型的线程
    // condition.await()方法来让当前类型的线程进入等待状态
    // condition.signalAll()方法，唤醒指定类型的线程。
    // 如果使用synchronized来作为同步手段，等待及唤醒使用wait(),notifyAll()。
    // 使用condition来精确指定线程，并同步时。使用await(),signalAll()来作为等待及唤醒方法。
    private Condition producer = lock.newCondition();
    private Condition consumer = lock.newCondition();


    public void put(T t) {
        try {
            //锁定
            lock.lock();
            /*
            如果使用if来判断线程是否需要wait(等待)，那么首先判断只判断一次，在被唤醒时就不会再作第二次判断了。
            而真实的应该是线程在每次的运行时都应该再作一次判断是否可以运行了，因为同时有多个线程时。
            前面的线程完成时，第二个线程在运行之前应该再作一次判断。是否可以执行，还是继续等待，让其它正确的线程运行。
             */
            while (lists.size() == MAX) {   //想想为什么是while而不是if
                producer.await();
            }
            lists.add(t);
            ++count;
            consumer.signalAll();   //通知消费者线程
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            //释放锁
            lock.unlock();
        }
    }

    public synchronized T get() {
        T t = null;
        try {
            lock.lock();
            while (lists.size() == 0) {
                consumer.await();
            }
            t = lists.remove(0);
            this.count--;
            producer.signalAll();   //通知生产者线程
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            //手动释放锁
            lock.unlock();
        }
        return t;
    }

    public static void main(String[] args) {

        MyCoutainer2<String> c = new MyCoutainer2<>();

        //启动消费者线程10个
        for (int i = 1; i < 11; i++) {
            new Thread(() -> {
                for (int j = 1; j < 6; j++) {
                    System.out.println(Thread.currentThread().getName() + "消费了" + c.get());
                }
            }, "消费者:" + i + "号, ").start();
        }

        try {
            TimeUnit.SECONDS.sleep(2);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        //启动生产者线程2个
        for (int i = 1; i < 3; i++) {
            new Thread(() -> {
                for (int j = 1; j < 26; j++) {
                    c.put(Thread.currentThread().getName() + "生产的第" + j + "个产品");
                }
            }, "生产者:" + i + "号").start();
        }
    }
}
